#!/usr/bin/env python
"""
开发环境WebSocket支持启动脚本
使用Daphne服务器支持WebSocket连接
"""
import os
import django
from django.core.management import execute_from_command_line

if __name__ == '__main__':
    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'core.settings')
    
    # 检查是否安装了daphne
    try:
        import daphne
        print("✅ Daphne已安装，支持WebSocket")
        
        # 使用daphne启动服务器
        from daphne.management.commands.runserver import Command as DaphneCommand
        
        # 设置Django
        django.setup()
        
        print("🚀 正在启动支持WebSocket的开发服务器...")
        print("📡 HTTP服务: http://127.0.0.1:8000")
        print("🔗 WebSocket服务: ws://127.0.0.1:8000/ws/")
        print("🛑 按Ctrl+C停止服务器")
        
        # 使用daphne启动
        os.system('daphne -p 8000 -b 127.0.0.1 core.asgi:application')
        
    except ImportError:
        print("❌ Daphne未安装")
        print("📦 正在安装daphne...")
        os.system('pip install daphne')
        print("✅ 安装完成，请重新运行此脚本")